package br.edu.ufcg.greengrid.model.machine.state.factory;

import java.util.HashMap;
import java.util.Map;

import br.edu.ufcg.greengrid.model.machine.state.GreenMachineState;
import br.edu.ufcg.greengrid.model.machine.state.GridMachineState;
import br.edu.ufcg.greengrid.model.machine.state.IdleMachineState;
import br.edu.ufcg.greengrid.model.machine.state.MachineState;
import br.edu.ufcg.greengrid.model.machine.state.UserMachineState;
import br.edu.ufcg.greengrid.model.machine.state.MachineState.MSEnum;

import static br.edu.ufcg.greengrid.model.machine.state.MachineState.MSEnum;

public class NormalMachineStateFactory implements MachineStateFactory {

	@Override
	public Map<MSEnum, MachineState> getStates() {
		Map<MSEnum, MachineState> states = new HashMap<MSEnum, MachineState>();
		states.put(MSEnum.GREEN, new GreenMachineState());
		states.put(MSEnum.GRID, new GridMachineState());
		states.put(MSEnum.IDLE, new IdleMachineState());
		states.put(MSEnum.USER, new UserMachineState());

		return states;
	}

}
